Python partition method